package com.shiyi.shiyiaiagent.rag;

import org.springframework.ai.chat.prompt.PromptTemplate;
import org.springframework.ai.rag.generation.augmentation.ContextualQueryAugmenter;

/**
 * 创建上下文查询增器的工厂
 */
public class LoveAppContextualQueryAugmenterFactory {

  public static ContextualQueryAugmenter create() {
    PromptTemplate promptTemplate = new PromptTemplate("""
      你应该输出以下内容，
      抱歉，我只能回答和恋爱相关的问题，别的没办法回答您。
      有问题可以联系客服人员
      """);
    return ContextualQueryAugmenter.builder()
      .allowEmptyContext(false)
      .emptyContextPromptTemplate(promptTemplate)
      .build();
  }
}
